
class Config:

    def __init__(self):
        self.train_raw_dataset_path = '../../dataset/train'
        self.train_dataset_path = '../../dataset/task2_train'
        self.batch_size = 128
        self.train_epoch = [5, 7, 9]
        self.train_epoch_vgg = [1, 3, 4]
        self.train_epoch_alexnet = [5, 7, 9]
        self.learning_rate = 1e-3
        self.learning_rate_vgg = 1e-2
        self.learning_rate_alexnet = 1e-2
        self.model_root_path = './model'

        self.raw_task2_test_path = '../../dataset/test_offline/task2'
        self.task2_test_path = '../../dataset/task2_test'
        self.task2_test_gt_path = '../../dataset/test_offline/task2_gt.json'


        self.task2_model_1 = 'epoch50_vgg.pth'
        # self.task2_model_1 = 'vgg16.pth'
        self.task2_model_2 = 'alexnet.pth'
        self.task3_model = 'test'